#-*- coding: UTF-8 -*-
__author__ = 'Jinkey'
from tgrocery import Grocery

#实例化贝叶斯分类器
grocery = Grocery('J_text_classify')

#引入训练集tuple，也可以引入txt文本
train_text = [
    ('段子','小明又来逗人了，笑得胃抽筋'),
    ('段子','惨不忍睹!没文化真是太可怕了!开心一刻'),
    ('段子','[爆笑]谁说穿丝袜不能放手机了'),
    ('段子','[段子]说的太好了!最好背下来!'),
    ('段子','开心秀 这么唠嗑，不被打死就不错了!'),
    ('段子','『嘻嘻哈哈』三个笑话，让你开心一整天'),
    ('养生','[安全用药]痛风发作 用药不可莽撞'),
    ('养生','你确定会煮面条?九个煮面技巧让你煮出润滑吸溜'),
    ('养生','一个价值连城的小处方：刮掉你的大肚子(男女通'),
    ('养生','泡脚可不是把脚往水里一放的事!实用方子拿走'),
    ('养生','男女睡觉前生吃这个，晚上想怎么任性就怎么任性'),
    ('养生','A不得不看的超强[防晒]食物!让你不再害怕阳光!'),
]

#开始训练机器
grocery.train(train_text)

#保存训练模型
grocery.save()

#加载训练模型
jinkeytext = Grocery('J_text_classify')
jinkeytext.load()

#预测文章分类
class_result = jinkeytext.predict('今晚吃什么比较健康')

#打印结果
print '该文章属于类别：' + str(class_result)